#!/bin/bash
#SBATCH -o ./example_4.out
#SBATCH -p gpu31,gpu32
#SBATCH --qos=normal
#SBATCH -J Jax
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH --ntasks-per-node=8
#SBATCH -t 200:00:00

python example_4.py --net_config=example_4_config.py \
                    --epoch_number=512 \
                    --batch_size=128 \
                    --learning_rate=0.1 \